package com.qf.repo.db;

import java.sql.*;

public class DBUtils {
    private static final String MYSQL_DRIVER = "com.mysql.cj.jdbc.Driver";
    private static final String MYSQL_URL = "jdbc:mysql://127.0.0.1:3306/repo";
    private static final String MYSQL_USER = "root";
    private static final String MYSQL_PWD = "root";
    private static Connection conn = null;
    private static PreparedStatement pst = null;

    /**
     * jdbc连接数据库库的步骤：
     *      1.加载驱动
     *      2.建立数据库的连接：url、user、password，返回数据库连接对象
     *      3.获取数据库预编译执行PreparedStatement(pst)对象
     *      4.执行数据库update(sql)、ResultSet query(sql)
     *      5.根据结果处理
     *      6.关闭连接、释放对象
     */
    static {
        try {
            Class.forName(MYSQL_DRIVER);//Class.forName() 是 Java 中的一个静态方法，用于动态加载和注册驱动程序类。
        } catch (ClassNotFoundException e) {
            throw new RuntimeException(e);
        }
    }

    /**
     * 该方法用于建立数据库的连接对象
     *
     * @return
     */
    public static Connection getConnection() {
        Connection connection = null;
        try {
            connection = DriverManager.getConnection(MYSQL_URL, MYSQL_USER, MYSQL_PWD);
        } catch (SQLException e) {
            throw new RuntimeException(e);
        }
        return connection;
    }

    /**
     * 该方法用于执行数据库操作，接受两个参数，第一个参数带有?的sql语句
     *
     * @param sql      具体要执行的sql语句
     * @param callBack 回调对象，用于调用者动态灵活的设置?的数据
     * @return
     */
    public static int executeUpdate(String sql, PstCallBack callBack) {
        conn = getConnection();
        int result = -1;
        try {
            if (conn != null) {
                //预编译执行sql语句
                pst = conn.prepareStatement(sql);
                //调用一系列的pst的 set方法给？占位符赋值
                callBack.pstSetValue(pst);
                result = pst.executeUpdate();
            }
        } catch (SQLException e) {
            throw new RuntimeException(e);
        } finally {
            close();
        }
        return result;
    }

    /**
     * 方法用于数据库的查询，会将数据库的查询结果统一返回
     *
     * @param sql         要执行的sql语句
     * @param pstCallBack 设置sql中的变量
     * @return
     */
    public static ResultSet query(String sql, PstCallBack pstCallBack) {
        ResultSet resultSet = null;
        conn = getConnection();
        try {
            pst = conn.prepareStatement(sql);
            //将pst滴哦献给回传给调用者，供调用者设置?(占位符)对应的数据
            if (pstCallBack != null) {
                pstCallBack.pstSetValue(pst);
            }
            resultSet = pst.executeQuery();
        } catch (SQLException e) {
            throw new RuntimeException(e);
        }
        return resultSet;
    }

    /**
     * 该方法用于关闭数据库连接
     */
    public static void close() {
        try {
            if (pst != null) {
                pst.close();
            }
            if (conn != null) {
                conn.close();
            }
        } catch (SQLException e) {
            throw new RuntimeException(e);
        }
    }
}
